package com.jim.healio.graphrag.model;

import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.AllArgsConstructor;

import java.util.Map;
import java.util.List;
import java.util.ArrayList;

@Data
@NoArgsConstructor
@AllArgsConstructor
public class DocumentSearchResult {
    
    /**
     * 文档ID
     */
    private String documentId;
    
    /**
     * 文档标题
     */
    private String title;
    
    /**
     * 文档内容片段
     */
    private String content;
    
    /**
     * 相似度分数 (0-1之间)
     */
    private double similarity;
    
    /**
     * 距离分数 (向量距离)
     */
    private double distance;
    
    /**
     * 文档元数据
     */
    private Map<String, Object> metadata;
    
    /**
     * 高亮匹配的文本片段
     */
    private String highlightedText;
    
    /**
     * 匹配的实体列表
     */
    private List<String> matchedEntities = new ArrayList<>();
    
    /**
     * 文档类型
     */
    private MedicalDocument.DocumentType documentType;
    
    /**
     * 文档来源
     */
    private String source;
    
    /**
     * 匹配的关键词
     */
    private List<String> matchedKeywords = new ArrayList<>();
    
    /**
     * 搜索排名
     */
    private int rank;
    
    /**
     * 便捷构造方法
     */
    public DocumentSearchResult(String documentId, String title, String content, double similarity) {
        this.documentId = documentId;
        this.title = title;
        this.content = content;
        this.similarity = similarity;
        this.distance = 1.0 - similarity; // 距离是相似度的补数
    }
    
    /**
     * 添加匹配的实体
     */
    public void addMatchedEntity(String entityId) {
        if (!this.matchedEntities.contains(entityId)) {
            this.matchedEntities.add(entityId);
        }
    }
    
    /**
     * 添加匹配的关键词
     */
    public void addMatchedKeyword(String keyword) {
        if (!this.matchedKeywords.contains(keyword)) {
            this.matchedKeywords.add(keyword);
        }
    }
    
    /**
     * 获取格式化的相似度百分比
     */
    public String getFormattedSimilarity() {
        return String.format("%.1f%%", similarity * 100);
    }
    
    /**
     * 是否为高相关性结果 (相似度 > 0.8)
     */
    public boolean isHighRelevance() {
        return similarity > 0.8;
    }
    
    /**
     * 是否为中等相关性结果 (0.6 < 相似度 <= 0.8)
     */
    public boolean isMediumRelevance() {
        return similarity > 0.6 && similarity <= 0.8;
    }
} 